package com.example.blog.utils;

import com.example.blog.constant.Constants;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.JwtParser;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.io.Decoders;
import io.jsonwebtoken.security.Keys;
import lombok.extern.slf4j.Slf4j;
import java.security.Key;
import java.util.Date;
import java.util.Map;

@Slf4j
public class JwtUtils {
    //设置过期时间1小时
    private static final long JWT_EXPIRATION = 60 * 60 * 1000;
    //密钥  长度有要求
    private static final String secretStr = "CIw6DzttvHA+XnrTa2B1EMhLoai1R0vC6jr0Q6y/qsU=";
    //生成 key
    private static final Key key = Keys.hmacShaKeyFor(Decoders.BASE64.decode(secretStr));

    /**
     * 生成token
     */
    public static String genJwtToken(Map<String, Object> claim) {
        //签名算法
        String token = Jwts.builder()
                .setClaims(claim) //自定义内容(负载)
                .setExpiration(new Date(System.currentTimeMillis() + JWT_EXPIRATION))   //设置过期时间
                .signWith(key) //设置签名
                .compact();

        //将生成的 token 返回
        return token;
    }

    /**
     * 校验token
     * Claims 为null: 表示校验失败
     */
    public static Claims parseToken(String token) {
        //创建解析器, 设置签名密钥
        JwtParser build = Jwts.parserBuilder().setSigningKey(key).build();
        Claims claims = null;
        try {
            //解析token
            claims = build.parseClaimsJws(token).getBody();
        } catch (Exception e) {
            //签名验证失败
            log.error("解析token失败, token:{}", token);
            return null;
        }
        return claims;
    }

    /**
     * 从token中获取用户id
     * @param token
     * @return
     */
    public static Integer getIdByToken(String token) {
        Claims claims = parseToken(token);
        if (claims != null) {
            Integer userId  = (Integer) claims.get(Constants.TOKEN_ID);
            if(userId > 0) {
                return userId;
            }
        }
        return null;
    }
}
